#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 19-6-23 上午11:46
# @Author  : Tom
# @Site    : 
# @File    : utils.py
# @Software: PyCharm

from rest_framework.pagination import PageNumberPagination
from rest_framework.response import Response


# 指定分页器
class PageNumber_User(PageNumberPagination):
    page_size = 10  # 在请求端没有指定单页的数据大小时，设置页面为5条数据
    # page_query_param 前端发送的页数关键字名，默认为"page"
    page_size_query_param = "pagesize"  # 前端发送的每页数目关键字名，默认为None
    max_page_size = 10

    # 重写返回的数据格式
    def get_paginated_response(self, data):
        return Response({
            'counts': self.page.paginator.count,
            'lists': data,
            'page': self.page.number,
            'pages': self.page.paginator.num_pages,
            'pagesize': self.page_size
        })
